package cloud.keveri.chat4j.api.entity.chat;

import cloud.keveri.chat4j.api.entity.Role;
import lombok.Data;

/**
 * 请求消息
 * 借鉴自  <a href="https://gitee.com/grt1228/chatgpt-java">chatgpt-java<a/>
 * <p>
 * 作者：沈自在
 **/
@Data
public class Message {

    /**
     * 角色
     */
    private String role;

    /**
     * 消息内容
     */
    private String content;

    /**
     * 名称
     */
    private String name;

    public Message(String role, String content) {
        this.role = role;
        this.content = content;
    }

    /**
     * 默认创建
     */
    public static Message of(String content) {
        return new Message(Role.USER.getValue(), content);
    }

    /**
     *
     */
    public static Message ofSystem(String content) {
        return new Message(Role.SYSTEM.getValue(), content);
    }

    /**
     *
     */
    public static Message ofAssistant(String content) {
        return new Message(Role.ASSISTANT.getValue(), content);
    }

}
